package date_202404

//135. 分发糖果
//困难
//相关标签
//相关企业
//n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
//
//你需要按照以下要求，给这些孩子分发糖果：
//
//每个孩子至少分配到 1 个糖果。
//相邻两个孩子评分更高的孩子会获得更多的糖果。
//请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。
//
//
//
//示例 1：
//
//输入：ratings = [1,0,2]
//输出：5
//解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
//示例 2：
//
//输入：ratings = [1,2,2]
//输出：4
//解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
//第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。
//
//
//提示：
//
//n == ratings.length
//1 <= n <= 2 * 104
//0 <= ratings[i] <= 2 * 104

func candy(ratings []int) int {
	if len(ratings) == 0 {
		return 0
	}
	if len(ratings) == 1 {
		return 1
	}

	levelIdxList := make([]int, 0, len(ratings))
	if ratings[0] <= ratings[1] {
		levelIdxList = append(levelIdxList, 0)
	}
	if ratings[len(ratings)-1] <= ratings[len(ratings)-2] {
		levelIdxList = append(levelIdxList, len(ratings)-1)
	}
	for i := 1; i < len(ratings)-1; i++ {
		if ratings[i-1] >= ratings[i] && ratings[i] <= ratings[i+1] {
			levelIdxList = append(levelIdxList, i)
		}
	}

	levelDepth := 1
	candies := make([]int, len(ratings))
	for len(levelIdxList) > 0 {
		nexLevelIdxList := make([]int, 0, 2*len(levelIdxList))
		for _, levelIdx := range levelIdxList {
			candies[levelIdx] = levelDepth
			if levelIdx > 0 && ratings[levelIdx-1] > ratings[levelIdx] {
				nexLevelIdxList = append(nexLevelIdxList, levelIdx-1)
			}
			if levelIdx < len(ratings)-1 && ratings[levelIdx+1] > ratings[levelIdx] {
				nexLevelIdxList = append(nexLevelIdxList, levelIdx+1)
			}
		}
		levelDepth = levelDepth + 1
		levelIdxList = nexLevelIdxList
	}

	sumCandy := 0
	for _, candyNum := range candies {
		sumCandy = sumCandy + candyNum
	}
	return sumCandy
}
